package com.tanhua.commons.template;

import com.baidu.aip.contentcensor.AipContentCensor;
import com.baidu.aip.contentcensor.EImgType;
import com.tanhua.commons.properties.BaiduCensorProperties;
import lombok.extern.slf4j.Slf4j;
import org.json.JSONObject;

import java.util.List;

/**
 * 百度内容审核
 */
@Slf4j
public class BaiduCensorTemplate {

    private AipContentCensor client;

    public BaiduCensorTemplate(BaiduCensorProperties properties) {
        client = new AipContentCensor(properties.getAppId(), properties.getAipKey(), properties.getAipToken());
        // 可选：设置网络连接参数
        client.setConnectionTimeoutInMillis(2000);
        client.setSocketTimeoutInMillis(60000);
    }

    /**
     * 图片内容审核，根据图片的字节数组审核
     * @param imageBytes 图片字节数组
     * @return 审核是否通过
     */
    public boolean imageCensor(byte[] imageBytes){
        //调用接口进行审核
        JSONObject response = client.imageCensorUserDefined(imageBytes, null);
		log.info("图片审核完成，结果:{}" + response);
        //返回结果：1合规，2不合规，3疑似，4审核失败
        return response.getInt("conclusionType") == 1;
    }

    /**
     * 审核图片内容，根据图片的url地址审核
     * @param url 图片的访问地址
     * @return 审核结果。true审核通过；false审核不通过
     */
    public boolean imageCensor(String url){
        JSONObject response = client.imageCensorUserDefined(url, EImgType.URL, null);
        log.info("图片审核完成, url:{}, 结果:{}", url, response);
        return response.getInt("conclusionType")==1;
    }

    /**
     * 批量审核图片
     * @param urlList 图片的url地址集合
     * @return true全部审核通过；false审核不通过
     */
    public boolean imageBatchCensor(List<String> urlList){
        for (String url : urlList) {
            JSONObject response = client.imageCensorUserDefined(url, EImgType.URL, null);
            //1合规，2不合规，3疑似，4审核失败
            if (response.getInt("conclusionType") != 1) {
                return false;
            }
        }

        return true;
    }
    
    /**
     * 审核文本内容
     * @param text 文本内容
     * @return 审核结果。true审核通过；false审核不通过
     */
    public boolean textCensor(String text) {
        // 调用接口进行审核
        JSONObject response = client.textCensorUserDefined(text);
		log.info("文本审核完成, text:{}, 结果:{}", text, response);
        // 返回结果 1合规，2不合规，3疑似，4审核失败
        return response.getInt("conclusionType") == 1;
    }

}